leetcode number of good pairs python